package com.atguigu.gulimall.order.config;

import com.atguigu.gulimall.order.entity.MqMessageEntity;
import com.atguigu.gulimall.order.service.MqMessageService;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.ConnectionFactory;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.support.converter.Jackson2JsonMessageConverter;
import org.springframework.amqp.support.converter.MessageConverter;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;

import java.util.Date;

/**
 * @Author : Jungle
 * @Description :
 */
@Configuration
public class MyRabbitConfig {
    //@Autowired
    private RabbitTemplate rabbitTemplate;
    @Autowired
    private MqMessageService mqMessageService;


    //参考  RabbitAutoConfiguration  这样配置会丢失一些配置信息
    @Primary
    @Bean
    public RabbitTemplate rabbitTemplate(ConnectionFactory connectionFactory) {
        RabbitTemplate template = new RabbitTemplate(connectionFactory);
        rabbitTemplate = template;
        template.setMessageConverter(messageConverter());
        initRabbitTemplate();
        return template;
    }

    /**
     * rabbitmq发送消息用json序列化,
     * 序列化后无需实现 Serializable  速度更快
     */
    @Bean
    public MessageConverter messageConverter() {
        return new Jackson2JsonMessageConverter();
    }

    /**
     * 1、只要抵达Broker代理，ack就会true
     * 定制RabbitTemplate
     * 1、服务器收到消息就回调
     * 1、设置yml  spring.rabbitmq.publisher-confirms=true
     * 2、设置确认回调  ConfirmCallback
     * 2、消息正确抵达队列进行回调
     * 1、spring.rabbitmq.publisher-returns=true
     * spring.rabbitmq.template.mandatory=true
     * 2、设置确认回调 ReturnCallback
     * <p>
     * 3、消费端确认（保证每个消息被正确消费，此时才可以broker删除）
     * spring.rabbitmq.listener.simple.acknowledge-mode=manual
     * 1、默认是自动确认模式，只要消息接收到，客户端会自动确认，服务端就会移除这个消息
     * 问题：
     * 我们收到很多消息，自动回复给服务器ack,只有一个消息处理成功宕机了。发生了消息丢失
     * 解决：
     * 消费者手动确认模式
     * 只要我们没有明确告诉MQ，货物被签收，没有Ack,消息就是一直是unacked状态，即使Consumer宕机，消息也不会丢失，会重新变为ready，
     * 下一次有新的consumer连接进来就发给他
     * 2、如何签收货物
     * 签收货物 channel.basicAck(deliveryTag,false);  业务成功完成就应该签收
     * 拒绝签收  channel.basicNack(deliveryTag,false,true);  业务失败就应该拒签
     */
    //@PostConstruct  // MyRabbitConfig对象创建完成来调用
    public void initRabbitTemplate() {
        //设置确认回调
        rabbitTemplate.setConfirmCallback(new RabbitTemplate.ConfirmCallback() {
            /**
             *
             * @param correlationData  当前消息的唯一关联数据（这个是消息额唯一id）
             * @param ack  消息是否成功收到
             * @param cause   失败的原因
             */
            /**
             * 1、做好消息确认机制（publisher.consumer[手动ack]）
             * 2、每一个发送的消息都在数据做好记录，定期将失败的消息再次发送一遍
             */
            @Override
            public void confirm(CorrelationData correlationData, boolean ack, String cause) {
                //服务器已经收到了
                //作为消息中间件，修改消息的状态 correlationData设置一个id,获取到消息的状态，保存到数据库
                //修改消息的状态
                System.out.println(
                        "confirm...correlationData[" + correlationData + "]==>" +
                                "ack[" + ack + "]==>" +
                                "cause[" + cause + "]==>");
                System.out.println("order服务ConfirmCallback回调");
            }
        });


        //设置消息抵达队列确认回调
        rabbitTemplate.setReturnCallback(new RabbitTemplate.ReturnCallback() {
            /**
             * 只要消息没有投递给指定的队列，就会触发这个失败回调
             * @param message  投递失败的消息详细信息
             * @param replyCode  回复状态码
             * @param replyText  回复的文本内容
             * @param exchange  当时这个消息发送给哪个交换机
             * @param routingKey  当时这个消息用哪个路由键
             */
            @Override
            public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {

                //保存失败消息
                MqMessageEntity messageEntity = new MqMessageEntity();
                String messageId = message.getMessageProperties().getMessageId();
                String string = message.getBody().getClass().toString();
                messageEntity.setMessageId(messageId)
                        .setContent(message.getBody().toString())
                        .setToExchane(exchange)
                        .setRoutingKey(routingKey)
                        .setClassType(string)
                        .setMessageStatus(replyCode)
                        .setCreateTime(new Date())
                        .setUpdateTime(new Date());
                mqMessageService.save(messageEntity);

                //报错误了，修改数据库当前消息的状态，--》错误
                System.out.println("Fail Message[" + message + "]==>" +
                        "replyCode[" + replyCode + "]==>" +
                        "replyText[" + replyText + "]==>" +
                        "exchange[" + exchange + "]==>" +
                        "routingKey[" + routingKey + "]==>" + "已经保存order服务的oms_mq_message表中");

                System.out.println("order服务ReturnCallback回调");
            }
        });


        //void confirm(@Nullable CorrelationData correlationData, boolean ack, @Nullable String cause);
        //rabbitTemplate.setConfirmCallback((correlationData,ack,cause)->{
        //    System.out.println();
        //});

    }
}
